package com.duoduo.树;

import com.duoduo.TreeNode;

import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
import java.util.List;

/**
 * @author dl.chai
 * @version 1.0
 * @description: 二叉树展开为链表114
 * 给你二叉树的根结点 root ，请你将它展开为一个单链表：
 *
 * 展开后的单链表应该同样使用 TreeNode ，其中 right 子指针指向链表中下一个结点，而左子指针始终为 null 。
 * 展开后的单链表应该与二叉树 先序遍历 顺序相同。
 *
 *
 * 示例 1：
 *
 *
 * 输入：root = [1,2,5,3,4,null,6]
 * 输出：[1,null,2,null,3,null,4,null,5,null,6]
 * 示例 2：
 *
 * 输入：root = []
 * 输出：[]
 * @date 2023/3/15 19:51
 */

public class 二叉树展开为链表114 {
    public void flatten(TreeNode root) {
        //从根节点开始
        while (root != null) {
            //如果根节点的左节点为空，继续寻找下一个
            if (root.left == null) {
                root = root.right;
            } else {
                TreeNode pre = root.left;
                //寻找左子树的最右子树
                while (pre.right != null) {
                    pre = pre.right;
                }
                //将根节点的右子树接到当前节点的右子树上
                pre.right = root.right;
                //将根节点的左子树换为右子树
                root.right = root.left;
                //根节点左子树置空
                root.left = null;
                //继续遍历
                root = root.right;
            }
        }
    }
    // 前序遍历可以通过递归或者迭代的方式实现。以下代码通过递归实现前序遍历。
    public void flatten2(TreeNode root) {
        List<TreeNode> list = new ArrayList<TreeNode>();
        preorderTraversal(root, list);
        int size = list.size();
        for (int i = 1; i < size; i++) {
            TreeNode prev = list.get(i - 1), curr = list.get(i);
            prev.left = null;
            prev.right = curr;
        }
    }

    public void preorderTraversal(TreeNode root, List<TreeNode> list) {
        if (root != null) {
            list.add(root);
            preorderTraversal(root.left, list);
            preorderTraversal(root.right, list);
        }
    }
    // 以下代码通过迭代实现前序遍历。
    public void flatten3(TreeNode root) {
        List<TreeNode> list = new ArrayList<TreeNode>();
        Deque<TreeNode> stack = new LinkedList<TreeNode>();
        TreeNode node = root;
        while (node != null || !stack.isEmpty()) {
            while (node != null) {
                list.add(node);
                stack.push(node);
                node = node.left;
            }
            node = stack.pop();
            node = node.right;
        }
        int size = list.size();
        for (int i = 1; i < size; i++) {
            TreeNode prev = list.get(i - 1), curr = list.get(i);
            prev.left = null;
            prev.right = curr;
        }
    }

}
